/*
 * Copyright 2016 L4 Digital LLC. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package remix.myplayer.ui.widget.fastcroll_recyclerview;

import android.content.Context;
import android.support.annotation.ColorInt;
import android.support.v7.widget.RecyclerView;
import android.util.AttributeSet;
import android.view.ViewGroup;
import android.view.ViewParent;


public class FastScrollRecyclerView extends RecyclerView {

  private FastScroller mFastScroller;

  public FastScrollRecyclerView(Context context) {
    super(context);
    layout(context, null);
    setLayoutParams(new LayoutParams(LayoutParams.MATCH_PARENT, LayoutParams.WRAP_CONTENT));
  }

  public FastScrollRecyclerView(Context context, AttributeSet attrs) {
    this(context, attrs, 0);
  }

  public FastScrollRecyclerView(Context context, AttributeSet attrs, int defStyleAttr) {
    super(context, attrs, defStyleAttr);
    layout(context, attrs);
  }

  @Override
  public void setAdapter(Adapter adapter) {
    super.setAdapter(adapter);

    if (adapter instanceof FastScroller.SectionIndexer) {
      mFastScroller.setSectionIndexer((FastScroller.SectionIndexer) adapter);
    } else {
      mFastScroller.setSectionIndexer(null);
    }
  }

  /**
   * Set the enabled state of fast scrolling.
   *
   * @param enabled True to enable fast scrolling, false otherwise
   */
  public void setFastScrollEnabled(boolean enabled) {
    mFastScroller.setEnabled(enabled);
  }

  /**
   * Hide the scrollbar when not scrolling.
   *
   * @param hideScrollbar True to hide the scrollbar, false to show
   */
  public void setHideScrollbar(boolean hideScrollbar) {
    mFastScroller.setHideScrollbar(hideScrollbar);
  }

  /**
   * Display a scroll track while scrolling.
   *
   * @param visible True to show scroll track, false to hide
   */
  public void setTrackVisible(boolean visible) {
    mFastScroller.setTrackVisible(visible);
  }

  /**
   * Set the color of the scroll track.
   *
   * @param color The color for the scroll track
   */
  public void setTrackColor(@ColorInt int color) {
    mFastScroller.setTrackColor(color);
  }

  /**
   * Set the color for the scroll handle.
   *
   * @param color The color for the scroll handle
   */
  public void setHandleColor(@ColorInt int color) {
    mFastScroller.setHandleColor(color);
  }

  /**
   * Set the background color of the index bubble.
   *
   * @param color The background color for the index bubble
   */
  public void setBubbleColor(@ColorInt int color) {
    mFastScroller.setBubbleColor(color);
  }

  /**
   * Set the text color of the index bubble.
   *
   * @param color The text color for the index bubble
   */
  public void setBubbleTextColor(@ColorInt int color) {
    mFastScroller.setBubbleTextColor(color);
  }

  /**
   * Set the fast scroll state change listener.
   *
   * @param fastScrollStateChangeListener The interface that will listen to fastscroll state change events
   */
  public void setFastScrollStateChangeListener(
      FastScrollStateChangeListener fastScrollStateChangeListener) {
    mFastScroller.setFastScrollStateChangeListener(fastScrollStateChangeListener);
  }

  @Override
  protected void onAttachedToWindow() {
    super.onAttachedToWindow();
    mFastScroller.attachRecyclerView(this);

    ViewParent parent = getParent();

    if (parent instanceof ViewGroup) {
      ViewGroup viewGroup = (ViewGroup) parent;
      viewGroup.removeView(mFastScroller);
      viewGroup.addView(mFastScroller);
      mFastScroller.setLayoutParams(viewGroup);
    }
  }

  @Override
  protected void onDetachedFromWindow() {
    mFastScroller.detachRecyclerView();
    super.onDetachedFromWindow();
  }

  private void layout(Context context, AttributeSet attrs) {
    mFastScroller = new FastScroller(context, attrs);
  }
}
